package dao;

import org.apache.commons.dbutils.QueryRunner;
import org.apache.commons.dbutils.ResultSetHandler;
import org.apache.commons.dbutils.handlers.BeanHandler;
import org.apache.commons.dbutils.handlers.BeanListHandler;
import org.apache.commons.dbutils.handlers.ScalarHandler;
import utils.JDBCUtil;

import java.sql.Connection;
import java.sql.ResultSet;
import java.sql.SQLException;
import java.util.ArrayList;
import java.util.List;

public class BaseDao {
    protected QueryRunner queryRunner = new QueryRunner();

    /**
     * DML
     *
     * @return -1:更新失败 other:受影响的行数
     */
    public int update(String sql, Object... args) {
        Connection conn = JDBCUtil.getConnection();
        try {
            return queryRunner.update(conn, sql, args);
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            JDBCUtil.release(conn);
        }
        return -1;
    }

    /**
     * 查询返回一个java bean对象
     *
     * @param type 返回的对象的类型
     * @param sql  执行的SQL语句
     * @param args SQL语句所需参数
     * @param <T>  返回对象类型的泛型
     * @return T
     */
    public <T> T queryForOne(Class<T> type, String sql, Object... args) {
        Connection conn = JDBCUtil.getConnection();
        try {
            return queryRunner.query(conn, sql, new BeanHandler<T>(type), args);
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            JDBCUtil.release(conn);
        }
        return null;
    }

    /**
     * 查询返回多个java bean对象
     *
     * @param type 返回的对象的类型
     * @param sql  执行的SQL语句
     * @param args SQL语句所需参数
     * @param <T>  返回对象类型的泛型
     * @return List
     */
    public <T> List<T> queryForList(Class<T> type, String sql, Object... args) {
        Connection conn = JDBCUtil.getConnection();
        try {
            return queryRunner.query(conn, sql, new BeanListHandler<T>(type), args);
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            JDBCUtil.release(conn);
        }
        return null;
    }

    /**
     * 查询返回单个值，主要针对统计函数
     *
     * @param sql  执行的SQL语句
     * @param args SQL语句所需参数
     * @return
     */
    public Object queryForSingleValue(String sql, Object... args) {
        Connection conn = JDBCUtil.getConnection();
        try {
            return queryRunner.query(conn, sql, new ScalarHandler(), args);
        } catch (SQLException e) {
            e.printStackTrace();
        } finally {
            JDBCUtil.release(conn);
        }
        return null;
    }


}
